from llm import NovelLLM
from util import writeFile
from .base import BaseNode

class Step2Node(BaseNode):
    def run(self, state):
        prompt = """
# Task
一段式概括，按“三幕式结构”组织成一个段落，包含背景、三次灾难事件和结局。
1. 背景与一到两个主角介绍；
2. 首个灾难迫使主角行动；
3. 次灾难改变主角行为模式；
4. 第三次灾难推动结局；
5. 最终对抗与结局

# 当前故事梗概
{summary}""".format(summary=state['step1_output'])

        result = NovelLLM().call_llm(prompt, caller_name="Step2")
        state['step2_output'] = result

        writeFile("output/step2.txt", result)

        print(f"Step2 完成段落扩展")
        return state